import dlib
print("dlib CUDA support: ", dlib.DLIB_USE_CUDA)

# 如果 CUDA 支持是 True，那么 GPU 应该能够使用
if dlib.DLIB_USE_CUDA:
    print("CUDA is available and will be used.")
    print(f"CUDA GPU Count: {dlib.cuda.get_num_devices()}")
    print(f"CUDA Device Name: {dlib.cuda.get_device_name(0)}")
else:
    print("CUDA is not available, using CPU.")
